package cn.zjyj.module.zjyj.service.risklist;

import cn.hutool.core.collection.CollUtil;
import cn.zjyj.module.zjyj.dal.dataobject.device.DeviceDO;
import cn.zjyj.module.zjyj.dal.dataobject.deviceinstance.DeviceInstanceDO;
import cn.zjyj.module.zjyj.dal.mysql.device.DeviceMapper;
import cn.zjyj.module.zjyj.dal.mysql.deviceinstance.DeviceInstanceMapper;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;

import org.springframework.validation.annotation.Validated;

import java.util.*;

import cn.zjyj.module.zjyj.controller.admin.risklist.vo.*;
import cn.zjyj.module.zjyj.dal.dataobject.risklist.RiskListDO;
import cn.zjyj.framework.common.common.pojo.PageResult;
import cn.zjyj.framework.common.common.util.object.BeanUtils;

import cn.zjyj.module.zjyj.dal.mysql.risklist.RiskListMapper;

import static cn.zjyj.framework.common.common.exception.util.ServiceExceptionUtil.exception;
import static cn.zjyj.framework.common.common.util.collection.CollectionUtils.convertList;
import static cn.zjyj.module.system.enums.ErrorCodeConstants.*;

/**
 * 风险清单 Service 实现类
 *
 * @author 芋道源码
 */
@Service
@Validated
public class RiskListServiceImpl implements RiskListService {

    @Resource
    private RiskListMapper riskListMapper;

    @Resource
    DeviceMapper deviceMapper;

    @Resource
    DeviceInstanceMapper deviceInstanceMapper;

    @Override
    public Long createRiskList(RiskListSaveReqVO createReqVO) {
        // 插入
        RiskListDO riskList = BeanUtils.toBean(createReqVO, RiskListDO.class);
        riskListMapper.insert(riskList);
        // 返回
        return riskList.getId();
    }

    @Override
    public void updateRiskList(RiskListSaveReqVO updateReqVO) {
        // 校验存在
        validateRiskListExists(updateReqVO.getId());
        // 更新
        RiskListDO updateObj = BeanUtils.toBean(updateReqVO, RiskListDO.class);
        riskListMapper.updateById(updateObj);
    }

    @Override
    public void deleteRiskList(Long id) {
        // 校验存在
        validateRiskListExists(id);
        // 删除
        riskListMapper.deleteById(id);
    }

    @Override
    public void deleteRiskListListByIds(List<Long> ids) {
        // 校验存在
        validateRiskListExists(ids);
        // 删除
        riskListMapper.deleteByIds(ids);
    }

    private void validateRiskListExists(List<Long> ids) {
        List<RiskListDO> list = riskListMapper.selectByIds(ids);
        if (CollUtil.isEmpty(list) || list.size() != ids.size()) {
            throw exception(RISK_LIST_NOT_EXISTS);
        }
    }

    private void validateRiskListExists(Long id) {
        if (riskListMapper.selectById(id) == null) {
            throw exception(RISK_LIST_NOT_EXISTS);
        }
    }

    @Override
    public RiskListDO getRiskList(Long id) {
        return riskListMapper.selectById(id);
    }

    @Override
    public PageResult<RiskListDO> getRiskListPage(RiskListPageReqVO pageReqVO) {
        return riskListMapper.selectPage(pageReqVO);
    }

    @Override
    public String getRiskName(Integer riskType, Long riskTypeInfoId) {
        if (riskType == 2) {
            DeviceDO deviceDO = deviceMapper.selectById(riskTypeInfoId);
            return deviceDO.getName();
        } else {
            RiskListDO riskListDO = riskListMapper.selectById(riskTypeInfoId);
            return riskListDO.getName();
        }
    }

    @Override
    public Long getInstanceSize(Integer riskType, Long riskTypeInfoId) {
        if (riskType == 2) {
            return deviceInstanceMapper.selectCount(DeviceInstanceDO::getDeviceId, riskTypeInfoId);
        } else {
            return 1L;
        }
    }
}